from shuangti.Item01.AIAssistant import AIAssistant


class TranslationAssistant(AIAssistant):
    """翻译助手子类，支持的语言对为自定义属性"""
    def __init__(self, aid: str, name: str, version: str, is_online: bool, ability_score: int, supported_languages: str):
        super().__init__(aid, name, "翻译助手", version, is_online, ability_score)
        self.supported_languages = supported_languages  # 自定义属性：支持的语言对（中-英、中-日等）

    def execute_task(self, task_content: str) -> str:
        """执行翻译任务（模拟中译英）"""
        if not self.is_online:
            return f"❌ {self.name} 当前离线，无法执行任务。"
        # 模拟翻译结果
        translation_example = "Input: 人工智能改变世界 → Output: AI changes the world"
        return (f"✅ {self.name}（翻译助手）已执行任务：\n任务内容：{task_content}\n"
                f"处理结果：{translation_example}\n支持语言对：{self.supported_languages}")

    def __str__(self) -> str:
        base_info = super().__str__()
        return f"{base_info}, 支持语言对: {self.supported_languages}"